import test_prompt
import argparse

if __name__ == "__main__":
    # 创建参数解析器
    parser = argparse.ArgumentParser(description='综述生成')
    
    # 添加命令行参数
    parser.add_argument('--topic', type=str, required=True, help='指定主题')
    parser.add_argument('--token', type=str, required=False, help='指定API地址')
    parser.add_argument('--local', type=bool, required=False, help='是否使用本地API')
    
    # 解析命令行参数
    args = parser.parse_args()
    
    # 使用参数
    topic = args.topic
    token = args.token
    local = args.local
    if local is None:
        local = False
    if token:
        test_prompt.test_keyword_expand(topic, file_path="review.md", base="results", api_key=token)
    else:
        test_prompt.test_keyword_expand(topic, file_path="review.md", base="results", local=local)
    